//给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。
//
//
//
// 示例 1：
//
//
//输入：nums = [1,1,1], k = 2
//输出：2
//
//
// 示例 2：
//
//
//输入：nums = [1,2,3], k = 3
//输出：2
//
//
//
//
// 提示：
//
//
// 1 <= nums.length <= 2 * 10⁴
// -1000 <= nums[i] <= 1000
// -10⁷ <= k <= 10⁷
//
// Related Topics 数组 哈希表 前缀和 👍 1303 👎 0

package leetcode.editor.cn;

import java.util.HashMap;

@SuppressWarnings("all")
//Java：和为 K 的子数组
class 和为K的子数组 {
    public static void main(String[] args) {
        Solution solution = new 和为K的子数组().new Solution();
        // TO TEST
        int[] nums = {1, 1, 1};
        System.out.println(solution.subarraySum(nums, 2));
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        public int subarraySum(int[] nums, int k) {

            HashMap<Integer, Integer> prefix = new HashMap<>();
            prefix.put(0, 1);
            int count = 0;
            int crrPrefixSum = 0;
            for (int i = 0; i < nums.length; i++) {
                crrPrefixSum += nums[i];
                count += prefix.getOrDefault(crrPrefixSum - k, 0);
                prefix.put(crrPrefixSum, prefix.getOrDefault(crrPrefixSum, 0) + 1);
            }

            return count;
        }

        //time out
        private int dp(int[] nums, int k) {
            int dp[][] = new int[nums.length][nums.length];
            int count = 0;
            for (int i = dp.length - 1; i >= 0; i--) {
                for (int j = i; j < dp.length; j++) {
                    if (i == j)
                        dp[i][j] = nums[i];
                    else
                        dp[i][j] = dp[i + 1][j] + nums[i];
                    if (dp[i][j] == k)
                        count++;
                }
            }
            return count;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)


}
